package cn.yilongquan.util;

import io.minio.MinioClient;
import io.minio.PutObjectArgs;
import io.minio.RemoveObjectArgs;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;

import javax.annotation.PostConstruct;
import java.io.InputStream;
import java.util.UUID;

/**
 * MinIO 工具类，封装文件上传、删除等操作
 */
@Component // 标记为Spring组件，可被注入使用
public class MinioUtil {

    // 从配置文件读取MinIO参数（替代硬编码，更灵活）
    @Value("${minio.access-key}")
    private String accessKey;

    @Value("${minio.secret-key}")
    private String secretKey;

    @Value("${minio.endpoint}")
    private String endpoint;

    @Value("${minio.bucket-name}")
    private String bucketName;

    // MinIO客户端实例（全局复用）
    private MinioClient minioClient;

    // 初始化MinIO客户端（在Bean创建后执行）
    @PostConstruct
    public void init() {
        try {
            minioClient = MinioClient.builder()
                    .credentials(accessKey, secretKey)
                    .endpoint(endpoint)
                    .build();
        } catch (Exception e) {
            throw new RuntimeException("初始化MinIO客户端失败：" + e.getMessage());
        }
    }

    /**
     * 上传文件到MinIO
     * @param file 待上传的文件
     * @return 文件在MinIO中的访问URL
     */
    public String uploadFile(MultipartFile file) throws Exception {
        // 处理文件名：生成唯一文件名（避免重复），保留原文件后缀
        String originalFilename = file.getOriginalFilename();
        String suffix = originalFilename.substring(originalFilename.lastIndexOf(".")); // 如 .jpg
        String fileName = UUID.randomUUID().toString() + suffix; // 用UUID确保唯一

        // 调用MinIO客户端上传
        PutObjectArgs putObjectArgs = PutObjectArgs.builder()
                .bucket(bucketName) // 存储桶名称
                .object(fileName) // 文件名（可加目录，如 "images/xxx.jpg"）
                .stream(file.getInputStream(), file.getSize(), -1) // 文件流、大小，-1表示不分片
                .contentType(file.getContentType()) // 设置文件类型（如 image/jpeg）
                .build();

        minioClient.putObject(putObjectArgs);

        // 生成访问URL（格式：endpoint/bucketName/fileName）
        return endpoint + "/" + bucketName + "/" + fileName;
    }

    /**
     * 从MinIO删除文件
     * @param fileName 要删除的文件名（需与上传时的fileName一致）
     */
    public void deleteFile(String fileName) throws Exception {
        RemoveObjectArgs removeObjectArgs = RemoveObjectArgs.builder()
                .bucket(bucketName)
                .object(fileName)
                .build();
        minioClient.removeObject(removeObjectArgs);
    }

    /**
     * 下载文件（获取文件输入流）
     * @param fileName 要下载的文件名
     * @return 文件输入流（需自行关闭）
     */
    public InputStream downloadFile(String fileName) throws Exception {
        return minioClient.getObject(
                io.minio.GetObjectArgs.builder()
                        .bucket(bucketName)
                        .object(fileName)
                        .build()
        );
    }
}